import pandas as pd
import os

# --------------------------
# 用户配置部分（请根据实际情况修改）
# --------------------------

# 原始Excel文件路径
input_file = r'C:\Users\admin\Desktop\5.6语言处理\第一次处理\1.xlsx'

# 输出目录（拆分后的文件都放在这里）
output_dir = r'C:\Users\admin\Desktop\5.6语言处理\第二次处理'

# 基础信息列（A-F、R-V），共11列
base_columns = [
    '模块', '功能-页面', '图片', '中文', '说明', '标识',
    '平台', '状态', '最后编辑日期', '语言', '备注'
]

# 所有语言列（H-Q，共10个语言）
language_columns = {
    '意大利语': '意大利语',
    '法语': '法语',
    '日语': '日语',
    '德语': '德语',
    '西班牙语': '西班牙语',
    '俄语': '俄语',
    '葡萄牙语': '葡萄牙语',
    '阿拉伯语': '阿拉伯语',
    '繁体': '繁体',
    '简体': '简体'
}

# 确保输出目录存在
os.makedirs(output_dir, exist_ok=True)

# --------------------------
# 1. 读取原始数据并添加 id 列
# --------------------------

df = pd.read_excel(input_file, sheet_name='Sheet1')

# 添加 id 列（从1开始）
df.reset_index(drop=True, inplace=True)
df['id'] = df.index + 1

# 重新排列列顺序，确保 id 在最前面
cols = ['id'] + [col for col in df.columns if col != 'id']
df = df[cols]

# --------------------------
# 2. 保存基础信息文件
# --------------------------

base_df = df[['id'] + base_columns]
base_output_path = os.path.join(output_dir, '基础信息.xlsx')
base_df.to_excel(base_output_path, index=False)
print(f"✅ 基础信息已保存至: {base_output_path}")

# --------------------------
# 3. 逐个提取语言列，生成翻译模板文件
# --------------------------

for lang_title, lang_col in language_columns.items():
    # 构建当前语言的数据
    lang_df = df[['id', '英语', lang_col]].copy()
    lang_df.rename(columns={lang_col: f'{lang_title}-未修改'}, inplace=True)
    lang_df[f'{lang_title}-修改后'] = ''  # 新增空列

    # 重新排序列
    lang_df = lang_df[['id', '英语', f'{lang_title}-未修改', f'{lang_title}-修改后']]

    # 保存文件
    lang_output_path = os.path.join(output_dir, f'{lang_title}.xlsx')
    lang_df.to_excel(lang_output_path, index=False)
    print(f"✅ 已生成语言文件: {lang_output_path}")

print("🎉 第一步完成：ID列添加成功，所有文件已拆分完成！")